#pragma once

#include "iostream"
#include "vector"
#include "algorithm"
#include "semaphore.h"

using namespace std;
/*HJJ QQ479287006
 *现有函数 printNumber 可以用一个整数参数调用，并输出该整数到控制台。

例如，调用 printNumber(7) 将会输出 7 到控制台。
给你类 ZeroEvenOdd 的一个实例，该类中有三个函数：zero、even 和 odd 。ZeroEvenOdd 的相同实例将会传递给三个不同线程：

线程 A：调用 zero() ，只输出 0
线程 B：调用 even() ，只输出偶数
线程 C：调用 odd() ，只输出奇数
修改给出的类，以输出序列 "010203040506..." ，其中序列的长度必须为 2n 。

实现 ZeroEvenOdd 类：

ZeroEvenOdd(int n) 用数字 n 初始化对象，表示需要输出的数。
void zero(printNumber) 调用 printNumber 以输出一个 0 。
void even(printNumber) 调用printNumber 以输出偶数。
void odd(printNumber) 调用 printNumber 以输出奇数。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/print-zero-even-odd
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 * */

int n1;
sem_t t1;
sem_t t2;
sem_t t0;

ZeroEvenOdd(int n) {
    this->n = n;
    sem_init(&t0, 0, 1);
    sem_init(&t1, 0, 0);
    sem_init(&t2, 0, 0);
}

// printNumber(x) outputs "x", where x is an integer.
void zero(function<void(int)> printNumber) {
    for (int i = 0; i < n; i++) {
        sem_wait(&t0);
        printNumber(0);
        if (i % 2 == 0) {
            sem_post(&t1);
        } else {
            sem_post(&t2);
        }


    }

}

//偶数
void even(function<void(int)> printNumber) {


    for (int i = 1; i <= n; i++) {
        if (i % 2 == 0) {

            sem_wait(&t2);
            printNumber(i);
            sem_post(&t0);

        }

    }


}

//奇数
void odd(function<void(int)> printNumber) {

    for (int i = 1; i <= n; i++) {
        if (i % 2 != 0) {
            sem_wait(&t1);
            printNumber(i);
            sem_post(&t0);

        }

    }


}
